package xiaohong.leetcode.secondMonth;

import java.util.Arrays;

import static xiaohong.sort.RadixSortDemo.radixSort;

/**
 * <p>
 * 给你一个有序数组 nums ，请你 原地 删除重复出现的元素，使得出现次数超过两次的元素只出现两次 ，返回删除后数组的新长度。
 * 不要使用额外的数组空间，你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
 * <p>
 * 说明：
 * 为什么返回数值是整数，但输出的答案是数组呢？
 * 请注意，输入数组是以「引用」方式传递的，这意味着在函数里修改输入数组对于调用者是可见的。
 * 你可以想象内部操作如下:
 * // nums 是以“引用”方式传递的。也就是说，不对实参做任何拷贝
 * int len = removeDuplicates(nums);
 * // 在函数里修改输入数组对于调用者是可见的。
 * // 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。
 * for (int i = 0; i < len; i++) {
 *     print(nums[i]);
 * }
 * <p>
 * 示例 1：
 * 输入：nums = [1,1,1,2,2,3]
 * 输出：5, nums = [1,1,2,2,3]
 * 解释：函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3。 不需要考虑数组中超出新长度后面的元素。
 * <p>
 * 示例 2：
 * 输入：nums = [0,0,1,1,1,1,2,3,3]
 * 输出：7, nums = [0,0,1,1,2,3,3]
 * 解释：函数应返回新长度 length = 7, 并且原数组的前七个元素被修改为 0, 0, 1, 1, 2, 3, 3。不需要考虑数组中超出新长度后面的元素。
 */
public class DeleteSortArray {

    public static void main(String[] args) {
        int[] nums = {0,0,1,1,1,1,2,3,3,4,4,4,4,4,4,4,4};
        //  排序
        radixSort(nums);
        System.out.println(Arrays.toString(nums));
        System.out.println("removeDuplicates(nums) = " + removeDuplicates(nums));
        System.out.println(Arrays.toString(nums));
    }

    // nums = [0,0,1,1,1,1,2,3,3]
    public static int removeDuplicates(int[] nums) {
        int size = nums.length;
        // 如果长度小于等于2直接返回
        if(size <= 2){
            return size;
        }

        int write = 0, curIndex = 1; // 有效区右边界，当前处理位置

        int preNum = nums[0], count = 1;    // 前一个数，  当前数出现次数

        while (curIndex < size) {
            int curNum = nums[curIndex];
            if (curNum != preNum) { // 当前数首次出现，有效数
                nums[++write] = nums[curIndex++]; // 发货到有效区，扩充有效区
                preNum = curNum; // 记录该数
                count = 1;
            } else if (count < 2) { // 当前数出现次数<2，有效数，发货到有效区，扩充有效区
                nums[++write] = nums[curIndex++]; // 发货到有效区，扩充有效区
                count++;
            } else { // 当前数出现次数已经>=2，垃圾数
                curIndex++; // 扩充垃圾区
            }
        }

        return write + 1;
    }

}
